package com.ruoyi.file.util

import com.ruoyi.file.router.FileCompressService
import cn.hutool.core.io.FileUtil
import cn.hutool.core.util.RandomUtil
import org.slf4j.LoggerFactory
import org.slf4j.MDC
import ws.schild.jave.*
import java.io.File

fun String.source(): File = FileUtil.file(this)
fun String.targetPath() =
    this.replace(FileUtil.getName(this), "${RandomUtil.randomNumbers(6)}.${FileUtil.getSuffix(this)}")

/**
 * 视频属性设置
 */
const val maxBitRate = 128000
const val maxSamplingRate = 44100
const val BIT_RATE = 800000
const val maxFrameRate = 20
const val maxWidth = 1080

private val log = LoggerFactory.getLogger(FileCompressService::class.java)

/**
 * 设置音频属性
 */
fun AudioInfo.audioAttributes() = AudioAttributes().apply {

    log.info("开始配置音频属性")

    // 设置通用编码格式
    setCodec("aac")
    // 设置最大值：比特率越高，清晰度/音质越好
    // 设置音频比特率,单位:b (比特率越高，清晰度/音质越好，当然文件也就越大 128000 = 128kb)
    if (this@audioAttributes.bitRate > maxBitRate) {
        setBitRate(maxBitRate)
    }
    // 设置重新编码的音频流中使用的声道数（1 =单声道，2 = 双声道（立体声））。如果未设置任何声道值，则编码器将选择默认值 0。
    setChannels(this@audioAttributes.channels)

    // 设置编码时候的音量值，未设置为0,如果256，则音量值不会改变
    // audio.setVolume(256);
    if (this@audioAttributes.samplingRate > maxSamplingRate) {
        setSamplingRate(maxSamplingRate)
    }
}

/**
 *  视频编码属性配置
 */
fun VideoInfo.videoAttributes() = VideoAttributes().apply {

    log.info("开始配置视频属性")

    // 设置通用编码格式
    setCodec("h264")
    // 设置音频比特率,单位:b (比特率越高，清晰度/音质越好，当然文件也就越大 800000 = 800kb)
    if (this@videoAttributes.bitRate > BIT_RATE) {
        setBitRate(BIT_RATE)
    }
    // 视频帧率：15 f / s 帧率越低，效果越差
    // 设置视频帧率（帧率越低，视频会出现断层，越高让人感觉越连续），
    // 视频帧率（Frame rate）是用于测量显示帧数的量度。所谓的测量单位为每秒显示帧数(Frames per Second，简：FPS）或“赫兹”（Hz）。
    if (this@videoAttributes.frameRate > maxFrameRate) {
        setFrameRate(maxFrameRate)
    }
    // 限制视频宽高
    val width = this@videoAttributes.size.width
    val height = this@videoAttributes.size.height
    if (width > maxWidth) {
        val rat = width.toFloat() / maxWidth
        setSize(VideoSize(maxWidth, (height / rat).toInt()))
    }
}

/**
 *
 * 保存视频File的路径，原视频名，压缩后视频名
 *
 */
fun String.compressionVideo(traceId: String): String =
    try {

        //存放链路id
        MDC.put("trace_uuid", traceId)
        log.info("开始进行视频转换")
        targetPath().also {
            MultimediaObject(source()).let { mediaObj ->
                Encoder().encode(
                    MultimediaObject(source()),
                    FileUtil.file(it),
                    EncodingAttributes().apply {

                        //输出格式
                        format = "mp4"

                        //音频属性配置
                        audioAttributes = mediaObj.info.audio.audioAttributes()

                        //视频属性配置
                        videoAttributes = mediaObj.info.video.videoAttributes()
                    })
            }
        }
    } finally {
        FileUtil.del(this@compressionVideo)
    }